poj 3259 Wormholes (bellman_ford)

/*
 John的农场里N块地，M条路连接两块地，W个虫洞，虫洞是一条单向路，
 会在你离开之前把你传送到目的地，就是当你过去的时候时间会倒退Ts。
 我们的任务是知道会不会在从某块地出发后又回来，看到了离开之前的自己。
 简化下，就是看图中有没有负权环。有的话就是可以，没有的话就是不可以了。
 
 
*/
# include <stdio.h>
# include <algorithm>
# include <string.h>
using namespace std;
# define inf 100010
struct node
{
    int u;
    int v;
    int c;
};
node map[6010];
int n;
int dis[2510];
int tot;
int bellman_ford()
{
    int i,j,flag;
    for(i=1; i<=n; i++)
        dis[i]=inf;

    for(i=1; i<=n; i++)
    {
        flag=0;
        for(j=0; j<tot; j++)
        {
            if(dis[map[j].v]>dis[map[j].u]+map[j].c)
            {
                dis[map[j].v]=dis[map[j].u]+map[j].c;
                flag=1;
            }
        }
        if(flag==0)
            break;
    }

    for(i=0; i<tot; i++)
    {
        if(dis[map[i].v]>dis[map[i].u]+map[i].c)
            return 1;
    }
    return 0;
}
int main()
{
    int t,m,a,b,cc,w,i;
    while(~scanf("%d",&t))
    {
        while(t--)
        {
            tot=0;
            scanf("%d%d%d",&n,&m,&w);
            for(i=0; i<m; i++)
            {
                scanf("%d%d%d",&a,&b,&cc);
                map[tot].u=a;
                map[tot].v=b;
                map[tot++].c=cc;
                map[tot].u=b;
                map[tot].v=a;
                map[tot++].c=cc;
            }
            for(i=0; i<w; i++)
            {
                scanf("%d%d%d",&a,&b,&cc);
                map[tot].u=a;
                map[tot].v=b;
                map[tot++].c=-cc;
            }
            if(bellman_ford())
                printf("YES\n");
            else
                printf("NO\n");
        }
    }
    return 0;
}